#ifndef KPEXTOPIMGF32C1FLIP_H
#define KPEXTOPIMGF32C1FLIP_H

#include "kpextVkContext.h"
#include "kpextBaseCompute.h"

namespace kpext {

/**
 * @brief 单通道32位单精度浮点数图像 水平/垂直 翻转操作
 */
class OpImgF32C1Flip : public BaseCompute{
public:
    explicit OpImgF32C1Flip(kpextVkContext * ctx);

    virtual ~OpImgF32C1Flip() override;

    /**
     * @brief  重新构建计算管线及相关资源
     * @param cols  输入二维图像的列数
     * @param rows  输入二维图像的行数
     * @param inbuf  输入缓冲
     * @param outbuf  输出缓冲
     * @param hflip  水平翻转
     * @param vflip  垂直翻转
     */
    virtual void rebuild(int cols,
                         int rows,
                         std::shared_ptr<BaseBuffer> inbuf,
                         std::shared_ptr<BaseBuffer> outbuf,
                         bool hflip = true,
                         bool vflip = false
                         );

protected:

    /**
     * @brief 计算着色器的work group size
     */
    std::array<uint32_t, 3> m_workGroupSize;

private:
    using BaseCompute::rebuild;

};  // class kpext::OpImgF32C1Flip

} // namespace kpext

#endif // KPEXTOPIMGF32C1FLIP_H
